{"role": "system", "content": "你是一个小红书作家，我输入一篇小红书推文，你给这个小红书起一个名字。"}
{"role": "user", "content": "小红书的内容"}
{"role": "assistant", "content": "小红书的标题"}
{"role": "user", "content": "小红书的内容"}

{"prompt": "推文内容1", "completion": "标题1"}
{"prompt": "推文内容2", "completion": "标题2"}

# 准备训练数据
messages = []
with open('posts.csv') as f:
    reader = csv.reader(f)
    for row in reader:
        messages.append(json.dumps({"prompt": rows['prompt'], "completion": row['completion']}))

# 写入训练数据到文件,文件必须是JSON Lines格式，https://jsonlines.readthedocs.io/en/latest/
with open('train.jsonl', 'a') as f:
    for message in messages:
        f.write(message + "\n")

# 上传训练数据到OpenAI
upload_file = openai.File.create(file=open("train.jsonl"), purpose='fine-tune', )
print(upload_file)

# 训练模型
fine_tune_response  = openai.FineTune.create(training_file='文件ID', model='davinci')
print(fine_tune_response)

# 获取模型信息
fine_tune_response = openai.FineTune.retrieve(id='模型ID')
print(fine_tune_response)

# 列出所有Fine Tune模型
fine_tune_models = openai.FineTune.list()
print(fine_tune_models)

# 测试模型效果
new_prompt = "村上春树说：有的人会用一辈子治愈童年，有的人会用童年治愈一辈子。了解自己身上的习惯，看透自己原生家庭的痛和乐。\n书名：《原生家庭》\n作者: （美）苏珊·福沃德博士 / （美）克雷格·巴克\n有毒家庭千千万，作者将其总结为“天下无不是的父母”、“不称职的父母”、“操控型的父母”、“酗酒型的父母”、“身体虐待型的父母”、“言语虐待型的父母”和“性虐待型的父母”，这些父母对孩子的负面模式持续存在（而不是偶尔的），始终支配着孩子的生活，影响着孩子与人的正常交际交往。\n当然，要判断自己的原生家庭是否有毒确实不是一件容易的事情，因而苏珊还提供了调查问卷，帮助读者进行自测。\n书中内容：\n如何判断自己的原生家庭种类\n有毒的父母会对孩子造成什么样的伤害？\n做才能摆脱原生家庭的负面影响？\n读完的感受：\n简单来说，看完苏珊的《原生家庭：如何修补自己的性格缺陷》真的帮助我认识到自己许久以来人际交往的通病产生的本质原因。\n发现自己，接纳自己，才会治愈自己。\n很有意义的一本书。希望可以帮助到更多需要帮助的宝子。"
answer = openai.Completion.create(
    model='模型名',
    prompt=new_prompt,
    max_tokens=1000
)
print(answer)